#!/usr/bin/env python
# -*- coding: utf-8 -*-
import os,sys
from xml.etree import ElementTree
from PIL import Image

def tree_to_dict(tree):
    d = {}
    for index, item in enumerate(tree):
        if item.tag == 'key':
            if tree[index+1].tag == 'string':
                d[item.text] = tree[index + 1].text
            elif tree[index + 1].tag == 'true':
                d[item.text] = True
            elif tree[index + 1].tag == 'false':
                d[item.text] = False
            elif tree[index+1].tag == 'dict':
                d[item.text] = tree_to_dict(tree[index+1])
    return d

def gen_png_from_plist(plist_filename, png_filename):
    print plist_filename
    file_path = plist_filename.replace('.plist', '')
    big_image = Image.open(png_filename)
    root = ElementTree.fromstring(open(plist_filename, 'r').read())
    plist_dict = tree_to_dict(root[0])
    to_list = lambda x: x.replace('{','').replace('}','').split(',')
    for k,v in plist_dict['frames'].items():
        rectlist = to_list(v['frame'])
        offset= to_list(v['offset'])

        width = int( rectlist[3] if v['rotated'] else rectlist[2] )
        height = int( rectlist[2] if v['rotated'] else rectlist[3] )
        box=( 
            int(rectlist[0]),
            int(rectlist[1]),
            int(rectlist[0]) + width,
            int(rectlist[1]) + height,
            )
        sizelist = [ int(x) for x in to_list(v['sourceSize'])]
        rect_on_big = big_image.crop(box)

        if v['rotated']:
            rect_on_big = rect_on_big.transpose(Image.ROTATE_90)

        result_image = Image.new('RGBA', sizelist, (0,0,0,0))
        if v['rotated']:
            result_box=(
                ( sizelist[0] - height )/2,
                ( sizelist[1] - width )/2,
                ( sizelist[0] + height )/2,
                ( sizelist[1] + width )/2
                )
        else:
            result_box=(
                ( sizelist[0] - width )/2,
                ( sizelist[1] - height )/2,
                ( sizelist[0] + width )/2,
                ( sizelist[1] + height )/2
                )

        print sizelist,result_box
        result_image.paste(rect_on_big, result_box, mask=0)
        file_path="/Users/kaixin/Desktop/主界面/"+file_path.split("/")[-1]
        if not os.path.isdir(file_path):
            os.mkdir(file_path)
        outfile = file_path+'/' + k
        result_image.save(outfile)
def upackPlistFile(map_path):
    for root, subdirs, files in os.walk(map_path):
        for plist_filename in files:
            if plist_filename.endswith(".plist"):
                png_filename = plist_filename.replace(".plist",".png")
                print plist_filename, png_filename
                print map_path+plist_filename, map_path+png_filename
                if os.path.exists(map_path+plist_filename) and os.path.exists(map_path+png_filename):
                    try:
                        gen_png_from_plist(map_path+plist_filename, map_path+png_filename)
                    except Exception as e:
                        print e
                else:
                    print "no files"
if __name__ == '__main__':
	# upackPlistFile("/Users/cuilibing/Stzb/stzb/cached_res/test/res/")
    upackPlistFile("/Users/kaixin/Desktop/主界面/")
    # gen_png_from_plist( "/Users/mengjie/Downloads/assets/gameResources/map/additionCity.plist", "/Users/mengjie/Downloads/assets/gameResources/map/additionCity.png" )